How to assign partitions to your consumers